#ifndef __EPOLL_SERVER_HPP__
#define __EPOLL_SERVER_HPP__

#include <iostream>
#include <string>
#include <functional>

#include "Epoll.hpp"
#include "Sock.hpp"
#include "log.hpp"
#include <cassert>
namespace ns_epoll
{
    static const int default_num=8080;//默认的端口号
    const static int gnum=64;
    using std::cout;
    using std::cin;
    using std::string;
    class EpollServer
    {
    public:

        using func_t = std::function<void(std::string)>;
        //将对于数据的使用暴露给外面 传进来什么 就怎么操作
        EpollServer(func_t HandlerRequset,const uint16_t& port=default_num)
            :_port(port)
            ,_HandlerRequset(HandlerRequset)
            ,_revs_num(gnum)
        {
            //0.由于在私有成员中添加了一个获得就绪时间的数组指针
            //需要开辟空间 以方便以后使用
            _revs=new struct epoll_event[_revs_num];

            //1. 创建_listensock
            _listensock=Sock::Socket();
            Sock::Bind(_listensock,_port);
            Sock::Listen(_listensock);
            //2. 创建epoll模型
            _epfd=Epoll::CreateEpoll();
            logMessage(DEBUG, "init success, listensock: %d, epfd: %d", _listensock, _epfd);
            //3.将_listensock加入epoll
            if (!Epoll::CtlEpoll(_epfd, EPOLL_CTL_ADD, _listensock, EPOLLIN)) exit(6);
            logMessage(DEBUG, "add listensock to epoll success.");

        }
        void Start()
        {
            int timeout = -1;
            //阻塞式等待
            while(true)
            {
                LoopOnce(timeout);
            }
        }

        //循环一次
        void LoopOnce(int timeout)
        {
            int n = Epoll::WaitEpoll(_epfd, _revs, _revs_num, timeout);
            //if(n == _revs_num) //扩容
            switch (n)
            {
            case 0:
                logMessage(DEBUG, "timeout..."); // 3, 4
                break;
            case -1:
                logMessage(WARNING, "epoll wait error: %s", strerror(errno));
                break;
            default:
                // 等待成功
                logMessage(DEBUG, "get a event");
                HandlerEvents(n);
                break;
            }
        }
        void HandlerEvents(int n)
        {
            assert(n>0);
            //走进来就一定有资源就绪了 看看是哪种事件
            //_revs里面存放着所有的就绪事件 我们挨着处理就可以
            for(int i=0;i<n;i++)
            {
                int sock=_revs[i].data.fd;
                uint32_t revents=_revs[i].events;
                //读时间就绪了
                if(revents & EPOLLIN)
                {
                    if(sock==_listensock)
                    {
                        Acceptr(sock);
                    }
                    else
                    {
                        Recver(sock);
                    }
                }
            }
        }
        void Acceptr(int listensock)
        {
            std::string clientip;
            uint16_t clientport;
            int sock=Sock::Accept(listensock,&clientip, &clientport);
            if(sock<0) 
            {
                logMessage(WARNING, "accept error!");
                return;
            }
            //不知道什么时候才会发送消息，接着让epoll帮我们等
            if( ! Epoll::CtlEpoll(_epfd,EPOLL_CTL_ADD,sock,EPOLLIN) ) return ;
            logMessage(DEBUG, "add new sock : %d to epoll success", sock);
        }
        void Recver(int sock)
        {
            char buffer[1024];
            ssize_t n=recv(sock,buffer,sizeof(buffer)-1,0);
            if(n>0)
            {
                buffer[n]=0;
                _HandlerRequset(buffer);//用外部传入的方法进行处理数据
            }
            else if(n==0)
            {
                // 1. 先在epoll中去掉对sock的关心
                bool res = Epoll::CtlEpoll(_epfd, EPOLL_CTL_DEL, sock, 0);
                assert(res);
                (void)res;
                // 2. 再close文件
                close(sock);
                logMessage(NORMAL, "client %d quit, me too...", sock);
            }
            else
            {
                // 1. 先在epoll中去掉对sock的关心
                bool res = Epoll::CtlEpoll(_epfd, EPOLL_CTL_DEL, sock, 0);
                assert(res);
                (void)res;
                // 2. 在close文件
                close(sock);
                logMessage(NORMAL, "client recv %d error, close error sock", sock);

            }
        }

        ~EpollServer()
        {
            if(_listensock>=0) close(_listensock);
            if(_epfd>=0) close(_epfd);
            if(_revs) delete[] _revs;
        }

    private:
        int _listensock;
        int _epfd;
        uint16_t _port;
        struct epoll_event *_revs;//用于一次性获取多个就绪事件的数组
        int _revs_num;//数组的大小
        func_t _HandlerRequset;
    };
}

#endif